package com.atguigu.yygh.codegenerator;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Property;
import org.apache.ibatis.annotations.Mapper;

import java.util.Collections;

/**
 * @package: com.baidu.codegenerator
 * @className: FastCodeGenerator
 * @author: lijing
 * @Description:
 * @date: 2025/4/15 16:52
 */
//修改配置然后运行 即可
public class FastCodeGenerator {
    public static void main(String[] args) {
        //获取当前项目路径
        String property = System.getProperty("user.dir");
        FastAutoGenerator.create("jdbc:mysql://192.168.47.4:3306/yygh_cmn?characterEncoding=utf-8&useSSL=false&serverTimezone=UTC", "root", "root")
                //全局配置
                .globalConfig(builder -> {
                    builder.author("lijing") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .dateType(DateType.ONLY_DATE) //设置事件策略
                            .commentDate("yyyy-MM-dd") // 注释日期
                            .disableOpenDir()     //生成完成后不打开文件夹
                            .outputDir("E:\\personCode\\尚硅谷\\项目\\尚医通\\yygh-parent\\mybatis-plus-generator3.5\\src\\main\\java"); // 指定输出目录

                })
                //包配置
                .packageConfig(builder -> {
                    builder.parent("com.atguigu.yygh") // 设置父包名 TODO 改成自己的包名
                            .moduleName("") // 设置父包模块名
                            .entity("entity")
                            .service("service")
                            .serviceImpl("service.impl")
                            .controller("controller")
                            .mapper("mapper")
                            .xml("xml");
//                            .pathInfo(Collections.singletonMap(OutputFile.xml,
//                                    property + "/src/main/java/com/baidu/fastgenerator/mapper/xml"));
                    // TODO 设置mapperXml生成路径
                })

                .strategyConfig(builder -> {
                    builder.addInclude("dict") // 设置需要生成的表名 TODO 改成自己的表名
                            .addFieldPrefix("")//指定表的哪些字段去掉
                            .serviceBuilder()//开始设置服务层
                            .formatServiceFileName("%sService")//这里整体为 %s（代表表名）+ 后面的Service
                            .formatServiceImplFileName("%sServiceImpl")
                            .entityBuilder()
                            .enableColumnConstant()//生成字段常量
                            .naming(NamingStrategy.underline_to_camel)//生成符合驼峰命名
                            .enableChainModel()//支持链式书写
                            .enableLombok()//支持lombok
                            //添加字段填充，添加到数据库的字段，字段常量，字段常量值，字段常量类型，字段常量注解
                            //TODO 这个没有实际测试。
                            .addTableFills(new Property("creat_time", FieldFill.INSERT))
                            .addTableFills(new Property("update_time",FieldFill.INSERT_UPDATE))
                            .logicDeletePropertyName("deleted")//指出逻辑删除 TODO 没有实际测试
                            .enableTableFieldAnnotation()//提供字段注解
                            .controllerBuilder()
                            .formatFileName("%sController")
                            .enableRestStyle()//生成@restcontroller 风格
                            .mapperBuilder()
                            .superClass(BaseMapper.class)//继承
                            .formatMapperFileName("%sMapper")
                            .formatXmlFileName("%sMapper")
                            .mapperAnnotation(Mapper.class);
                })
                // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .templateEngine(new FreemarkerTemplateEngine())
                .execute();
    }
}
